home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / lib / python2.5 / binhex.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2008-10-29  |  16KB  |  559 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.5)
  3.  
  4. '''Macintosh binhex compression/decompression.
  5.  
  6. easy interface:
  7. binhex(inputfilename, outputfilename)
  8. hexbin(inputfilename, outputfilename)
  9. '''
  10. import sys
  11. import os
  12. import struct
  13. import binascii
  14. __all__ = [
  15.     'binhex',
  16.     'hexbin',
  17.     'Error']
  18.  
  19. class Error(Exception):
  20.     pass
  21.  
  22. (_DID_HEADER, _DID_DATA, _DID_RSRC) = range(3)
  23. REASONABLY_LARGE = 32768
  24. LINELEN = 64
  25. RUNCHAR = chr(144)
  26.  
  27. try:
  28.     from Carbon.File import FSSpec, FInfo
  29.     from MacOS import openrf
  30.     
  31.     def getfileinfo(name):
  32.         finfo = FSSpec(name).FSpGetFInfo()
  33.         (dir, file) = os.path.split(name)
  34.         fp = open(name, 'rb')
  35.         fp.seek(0, 2)
  36.         dlen = fp.tell()
  37.         fp = openrf(name, '*rb')
  38.         fp.seek(0, 2)
  39.         rlen = fp.tell()
  40.         return (file, finfo, dlen, rlen)
  41.  
  42.     
  43.     def openrsrc(name, *mode):
  44.         if not mode:
  45.             mode = '*rb'
  46.         else:
  47.             mode = '*' + mode[0]
  48.         return openrf(name, mode)
  49.  
  50. except ImportError:
  51.     
  52.     class FInfo:
  53.         
  54.         def __init__(self):
  55.             self.Type = '????'
  56.             self.Creator = '????'
  57.             self.Flags = 0
  58.  
  59.  
  60.     
  61.     def getfileinfo(name):
  62.         finfo = FInfo()
  63.         fp = open(name)
  64.         data = open(name).read(256)
  65.         for c in data:
  66.             if not c.isspace():
  67.                 if c < ' ' or ord(c) > 127:
  68.                     break
  69.                     continue
  70.         else:
  71.             finfo.Type = 'TEXT'
  72.         fp.seek(0, 2)
  73.         dsize = fp.tell()
  74.         fp.close()
  75.         (dir, file) = os.path.split(name)
  76.         file = file.replace(':', '-', 1)
  77.         return (file, finfo, dsize, 0)
  78.  
  79.     
  80.     class openrsrc:
  81.         
  82.         def __init__(self, *args):
  83.             pass
  84.  
  85.         
  86.         def read(self, *args):
  87.             return ''
  88.  
  89.         
  90.         def write(self, *args):
  91.             pass
  92.  
  93.         
  94.         def close(self):
  95.             pass
  96.  
  97.  
  98.  
  99.  
  100. class _Hqxcoderengine:
  101.     '''Write data to the coder in 3-byte chunks'''
  102.     
  103.     def __init__(self, ofp):
  104.         self.ofp = ofp
  105.         self.data = ''
  106.         self.hqxdata = ''
  107.         self.linelen = LINELEN - 1
  108.  
  109.     
  110.     def write(self, data):
  111.         self.data = self.data + data
  112.         datalen = len(self.data)
  113.         todo = (datalen // 3) * 3
  114.         data = self.data[:todo]
  115.         self.data = self.data[todo:]
  116.         if not data:
  117.             return None
  118.         
  119.         self.hqxdata = self.hqxdata + binascii.b2a_hqx(data)
  120.         self._flush(0)
  121.  
  122.     
  123.     def _flush(self, force):
  124.         first = 0
  125.         while first <= len(self.hqxdata) - self.linelen:
  126.             last = first + self.linelen
  127.             self.ofp.write(self.hqxdata[first:last] + '\n')
  128.             self.linelen = LINELEN
  129.             first = last
  130.         self.hqxdata = self.hqxdata[first:]
  131.         if force:
  132.             self.ofp.write(self.hqxdata + ':\n')
  133.         
  134.  
  135.     
  136.     def close(self):
  137.         if self.data:
  138.             self.hqxdata = self.hqxdata + binascii.b2a_hqx(self.data)
  139.         
  140.         self._flush(1)
  141.         self.ofp.close()
  142.         del self.ofp
  143.  
  144.  
  145.  
  146. class _Rlecoderengine:
  147.     '''Write data to the RLE-coder in suitably large chunks'''
  148.     
  149.     def __init__(self, ofp):
  150.         self.ofp = ofp
  151.         self.data = ''
  152.  
  153.     
  154.     def write(self, data):
  155.         self.data = self.data + data
  156.         if len(self.data) < REASONABLY_LARGE:
  157.             return None
  158.         
  159.         rledata = binascii.rlecode_hqx(self.data)
  160.         self.ofp.write(rledata)
  161.         self.data = ''
  162.  
  163.     
  164.     def close(self):
  165.         if self.data:
  166.             rledata = binascii.rlecode_hqx(self.data)
  167.             self.ofp.write(rledata)
  168.         
  169.         self.ofp.close()
  170.         del self.ofp
  171.  
  172.  
  173.  
  174. class BinHex:
  175.     
  176.     def __init__(self, .1, ofp):
  177.         (name, finfo, dlen, rlen) = .1
  178.         if type(ofp) == type(''):
  179.             ofname = ofp
  180.             ofp = open(ofname, 'w')
  181.             if os.name == 'mac':
  182.                 fss = FSSpec(ofname)
  183.                 fss.SetCreatorType('BnHq', 'TEXT')
  184.             
  185.         
  186.         ofp.write('(This file must be converted with BinHex 4.0)\n\n:')
  187.         hqxer = _Hqxcoderengine(ofp)
  188.         self.ofp = _Rlecoderengine(hqxer)
  189.         self.crc = 0
  190.         if finfo is None:
  191.             finfo = FInfo()
  192.         
  193.         self.dlen = dlen
  194.         self.rlen = rlen
  195.         self._writeinfo(name, finfo)
  196.         self.state = _DID_HEADER
  197.  
  198.     
  199.     def _writeinfo(self, name, finfo):
  200.         nl = len(name)
  201.         if nl > 63:
  202.             raise Error, 'Filename too long'
  203.         
  204.         d = chr(nl) + name + '\x00'
  205.         d2 = finfo.Type + finfo.Creator
  206.         d3 = struct.pack('>h', finfo.Flags)
  207.         d4 = struct.pack('>ii', self.dlen, self.rlen)
  208.         info = d + d2 + d3 + d4
  209.         self._write(info)
  210.         self._writecrc()
  211.  
  212.     
  213.     def _write(self, data):
  214.         self.crc = binascii.crc_hqx(data, self.crc)
  215.         self.ofp.write(data)
  216.  
  217.     
  218.     def _writecrc(self):
  219.         if self.crc < 0:
  220.             fmt = '>h'
  221.         else:
  222.             fmt = '>H'
  223.         self.ofp.write(struct.pack(fmt, self.crc))
  224.         self.crc = 0
  225.  
  226.     
  227.     def write(self, data):
  228.         if self.state != _DID_HEADER:
  229.             raise Error, 'Writing data at the wrong time'
  230.         
  231.         self.dlen = self.dlen - len(data)
  232.         self._write(data)
  233.  
  234.     
  235.     def close_data(self):
  236.         if self.dlen != 0:
  237.             raise Error, 'Incorrect data size, diff=%r' % (self.rlen,)
  238.         
  239.         self._writecrc()
  240.         self.state = _DID_DATA
  241.  
  242.     
  243.     def write_rsrc(self, data):
  244.         if self.state < _DID_DATA:
  245.             self.close_data()
  246.         
  247.         if self.state != _DID_DATA:
  248.             raise Error, 'Writing resource data at the wrong time'
  249.         
  250.         self.rlen = self.rlen - len(data)
  251.         self._write(data)
  252.  
  253.     
  254.     def close(self):
  255.         if self.state < _DID_DATA:
  256.             self.close_data()
  257.         
  258.         if self.state != _DID_DATA:
  259.             raise Error, 'Close at the wrong time'
  260.         
  261.         if self.rlen != 0:
  262.             raise Error, 'Incorrect resource-datasize, diff=%r' % (self.rlen,)
  263.         
  264.         self._writecrc()
  265.         self.ofp.close()
  266.         self.state = None
  267.         del self.ofp
  268.  
  269.  
  270.  
  271. def binhex(inp, out):
  272.     '''(infilename, outfilename) - Create binhex-encoded copy of a file'''
  273.     finfo = getfileinfo(inp)
  274.     ofp = BinHex(finfo, out)
  275.     ifp = open(inp, 'rb')
  276.     while None:
  277.         d = ifp.read(128000)
  278.         if not d:
  279.             break
  280.         
  281.         continue
  282.         ofp.close_data()
  283.         ifp.close()
  284.         ifp = openrsrc(inp, 'rb')
  285.         while None:
  286.             d = ifp.read(128000)
  287.             if not d:
  288.                 break
  289.             
  290.             continue
  291.             ofp.close()
  292.             ifp.close()
  293.             return None
  294.  
  295.  
  296. class _Hqxdecoderengine:
  297.     '''Read data via the decoder in 4-byte chunks'''
  298.     
  299.     def __init__(self, ifp):
  300.         self.ifp = ifp
  301.         self.eof = 0
  302.  
  303.     
  304.     def read(self, totalwtd):
  305.         '''Read at least wtd bytes (or until EOF)'''
  306.         decdata = ''
  307.         wtd = totalwtd
  308.         while wtd > 0:
  309.             if self.eof:
  310.                 return decdata
  311.             
  312.             wtd = ((wtd + 2) // 3) * 4
  313.             data = self.ifp.read(wtd)
  314.             while None:
  315.                 
  316.                 try:
  317.                     (decdatacur, self.eof) = binascii.a2b_hqx(data)
  318.                 except binascii.Incomplete:
  319.                     pass
  320.  
  321.                 newdata = self.ifp.read(1)
  322.                 if not newdata:
  323.                     raise Error, 'Premature EOF on binhex file'
  324.                 
  325.                 data = data + newdata
  326.                 continue
  327.                 decdata = decdata + decdatacur
  328.                 wtd = totalwtd - len(decdata)
  329.                 if not decdata and not (self.eof):
  330.                     raise Error, 'Premature EOF on binhex file'
  331.                     continue
  332.         return decdata
  333.  
  334.     
  335.     def close(self):
  336.         self.ifp.close()
  337.  
  338.  
  339.  
  340. class _Rledecoderengine:
  341.     '''Read data via the RLE-coder'''
  342.     
  343.     def __init__(self, ifp):
  344.         self.ifp = ifp
  345.         self.pre_buffer = ''
  346.         self.post_buffer = ''
  347.         self.eof = 0
  348.  
  349.     
  350.     def read(self, wtd):
  351.         if wtd > len(self.post_buffer):
  352.             self._fill(wtd - len(self.post_buffer))
  353.         
  354.         rv = self.post_buffer[:wtd]
  355.         self.post_buffer = self.post_buffer[wtd:]
  356.         return rv
  357.  
  358.     
  359.     def _fill(self, wtd):
  360.         self.pre_buffer = self.pre_buffer + self.ifp.read(wtd + 4)
  361.         if self.ifp.eof:
  362.             self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer)
  363.             self.pre_buffer = ''
  364.             return None
  365.         
  366.         mark = len(self.pre_buffer)
  367.         if self.pre_buffer[-3:] == RUNCHAR + '\x00' + RUNCHAR:
  368.             mark = mark - 3
  369.         elif self.pre_buffer[-1] == RUNCHAR:
  370.             mark = mark - 2
  371.         elif self.pre_buffer[-2:] == RUNCHAR + '\x00':
  372.             mark = mark - 2
  373.         elif self.pre_buffer[-2] == RUNCHAR:
  374.             pass
  375.         else:
  376.             mark = mark - 1
  377.         self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer[:mark])
  378.         self.pre_buffer = self.pre_buffer[mark:]
  379.  
  380.     
  381.     def close(self):
  382.         self.ifp.close()
  383.  
  384.  
  385.  
  386. class HexBin:
  387.     
  388.     def __init__(self, ifp):
  389.         if type(ifp) == type(''):
  390.             ifp = open(ifp)
  391.         
  392.         while None:
  393.             ch = ifp.read(1)
  394.             if not ch:
  395.                 raise Error, 'No binhex data found'
  396.             
  397.             if ch == '\r':
  398.                 continue
  399.             
  400.             if ch == ':':
  401.                 break
  402.             
  403.             if ch != '\n':
  404.                 dummy = ifp.readline()
  405.                 continue
  406.             continue
  407.             hqxifp = _Hqxdecoderengine(ifp)
  408.             self.ifp = _Rledecoderengine(hqxifp)
  409.             self.crc = 0
  410.             self._readheader()
  411.             return None
  412.  
  413.     
  414.     def _read(self, len):
  415.         data = self.ifp.read(len)
  416.         self.crc = binascii.crc_hqx(data, self.crc)
  417.         return data
  418.  
  419.     
  420.     def _checkcrc(self):
  421.         filecrc = struct.unpack('>h', self.ifp.read(2))[0] & 65535
  422.         self.crc = self.crc & 65535
  423.         if filecrc != self.crc:
  424.             raise Error, 'CRC error, computed %x, read %x' % (self.crc, filecrc)
  425.         
  426.         self.crc = 0
  427.  
  428.     
  429.     def _readheader(self):
  430.         len = self._read(1)
  431.         fname = self._read(ord(len))
  432.         rest = self._read(19)
  433.         self._checkcrc()
  434.         type = rest[1:5]
  435.         creator = rest[5:9]
  436.         flags = struct.unpack('>h', rest[9:11])[0]
  437.         self.dlen = struct.unpack('>l', rest[11:15])[0]
  438.         self.rlen = struct.unpack('>l', rest[15:19])[0]
  439.         self.FName = fname
  440.         self.FInfo = FInfo()
  441.         self.FInfo.Creator = creator
  442.         self.FInfo.Type = type
  443.         self.FInfo.Flags = flags
  444.         self.state = _DID_HEADER
  445.  
  446.     
  447.     def read(self, *n):
  448.         if self.state != _DID_HEADER:
  449.             raise Error, 'Read data at wrong time'
  450.         
  451.         if n:
  452.             n = n[0]
  453.             n = min(n, self.dlen)
  454.         else:
  455.             n = self.dlen
  456.         rv = ''
  457.         while len(rv) < n:
  458.             rv = rv + self._read(n - len(rv))
  459.         self.dlen = self.dlen - n
  460.         return rv
  461.  
  462.     
  463.     def close_data(self):
  464.         if self.state != _DID_HEADER:
  465.             raise Error, 'close_data at wrong time'
  466.         
  467.         if self.dlen:
  468.             dummy = self._read(self.dlen)
  469.         
  470.         self._checkcrc()
  471.         self.state = _DID_DATA
  472.  
  473.     
  474.     def read_rsrc(self, *n):
  475.         if self.state == _DID_HEADER:
  476.             self.close_data()
  477.         
  478.         if self.state != _DID_DATA:
  479.             raise Error, 'Read resource data at wrong time'
  480.         
  481.         if n:
  482.             n = n[0]
  483.             n = min(n, self.rlen)
  484.         else:
  485.             n = self.rlen
  486.         self.rlen = self.rlen - n
  487.         return self._read(n)
  488.  
  489.     
  490.     def close(self):
  491.         if self.rlen:
  492.             dummy = self.read_rsrc(self.rlen)
  493.         
  494.         self._checkcrc()
  495.         self.state = _DID_RSRC
  496.         self.ifp.close()
  497.  
  498.  
  499.  
  500. def hexbin(inp, out):
  501.     '''(infilename, outfilename) - Decode binhexed file'''
  502.     ifp = HexBin(inp)
  503.     finfo = ifp.FInfo
  504.     if not out:
  505.         out = ifp.FName
  506.     
  507.     if os.name == 'mac':
  508.         ofss = FSSpec(out)
  509.         out = ofss.as_pathname()
  510.     
  511.     ofp = open(out, 'wb')
  512.     while None:
  513.         d = ifp.read(128000)
  514.         if not d:
  515.             break
  516.         
  517.         continue
  518.         ofp.close()
  519.         ifp.close_data()
  520.         d = ifp.read_rsrc(128000)
  521.         if d:
  522.             ofp = openrsrc(out, 'wb')
  523.             ofp.write(d)
  524.             while None:
  525.                 d = ifp.read_rsrc(128000)
  526.                 if not d:
  527.                     break
  528.                 
  529.                 continue
  530.                 ofp.close()
  531.     d
  532.     if os.name == 'mac':
  533.         nfinfo = ofss.GetFInfo()
  534.         nfinfo.Creator = finfo.Creator
  535.         nfinfo.Type = finfo.Type
  536.         nfinfo.Flags = finfo.Flags
  537.         ofss.SetFInfo(nfinfo)
  538.     
  539.     ifp.close()
  540.  
  541.  
  542. def _test():
  543.     if os.name == 'mac':
  544.         import macfs as macfs
  545.         (fss, ok) = macfs.PromptGetFile('File to convert:')
  546.         if not ok:
  547.             sys.exit(0)
  548.         
  549.         fname = fss.as_pathname()
  550.     else:
  551.         fname = sys.argv[1]
  552.     binhex(fname, fname + '.hqx')
  553.     hexbin(fname + '.hqx', fname + '.viahqx')
  554.     sys.exit(1)
  555.  
  556. if __name__ == '__main__':
  557.     _test()
  558.  
  559.